#!/usr/bin/env python
# -*- coding:utf-8 -*-
# Author:Zhangcl
from core import login
from core.admin import Admin_run
from core.users import User_run


class User(object):
    def __init__(self):
                pass

    def user_choise(self):
        choise_menu='''
        \033[1;31m
        ---------Welcome----------
        1. 普通用户登录
        2. 管理员登录
        3. 退出\033[0m
        '''
        menu_dict = {
            '1': 'users',
            '2': 'admin',
            '3': 'loginout'
        }

        run_dict = {
            'users': User_run,
            'admin': Admin_run
        }

        while True:
            print(choise_menu)
            login_choise = input('请选择>>>>').strip()
            if login_choise in menu_dict:
                if int(login_choise) == 3:
                    exit()
                else:
                    login_user = input('请输入账户>>>')
                    login_passwd = input('请输入密码>>>')
                    login_obj = login.User_login(login_user, login_passwd, menu_dict[login_choise])
                    run = login_obj.type_choise()
                    if run:  # 如果返回ture，说明账户密码验证正确，执行对应模块下面的程序
                        run_dict[run](login_user).menu()
            else:
                print('输入错误！')